package main

import (
	"Demo/ginEssential/common"
	"log"
	"os"

	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
	"github.com/spf13/viper"
)

func main() {
	InitConfig()
	db := common.InitDB()
	defer db.Close()
	r := gin.Default()
	r = CollectRoute(r)
	port := viper.GetString("server.port")
	if port != "" {
		panic(r.Run(":" + port))
	}
	panic(r.Run())
}

// InitConfig 初始化相关配置
func InitConfig() {
	workDir, _ := os.Getwd()                 //获取当前的工作目录
	viper.SetConfigName("application")       //设置读取的文件名
	viper.SetConfigType("yml")               //设置读取的文件类型
	viper.AddConfigPath(workDir + "/config") //设置读取的文件路径
	err := viper.ReadInConfig()
	if err != nil {
		log.Fatal(err)
	}

}
